// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef UI_COMPOSITOR_TEST_IN_PROCESS_CONTEXT_FACTORY_H_
#define UI_COMPOSITOR_TEST_IN_PROCESS_CONTEXT_FACTORY_H_

#include <stdint.h>

#include "base/macros.h"
#include "cc/test/test_gpu_memory_buffer_manager.h"
#include "cc/test/test_image_factory.h"
#include "cc/test/test_shared_bitmap_manager.h"
#include "cc/test/test_task_graph_runner.h"
#include "ui/compositor/compositor.h"

namespace base {
class Thread;
}

namespace cc {
class OnscreenDisplayClient;
class SurfaceManager;
}

namespace ui {
class InProcessContextProvider;

class InProcessContextFactory : public ContextFactory {
public:
    // surface_manager is owned by the creator of this and must outlive the
    // context factory.
    InProcessContextFactory(bool context_factory_for_test,
        cc::SurfaceManager* surface_manager);
    ~InProcessContextFactory() override;

    // If true (the default) an OutputSurface is created that does not display
    // anything. Set to false if you want to see results on the screen.
    void set_use_test_surface(bool use_test_surface)
    {
        use_test_surface_ = use_test_surface;
    }

    // ContextFactory implementation
    void CreateOutputSurface(base::WeakPtr<Compositor> compositor) override;

    scoped_ptr<Reflector> CreateReflector(Compositor* mirrored_compositor,
        Layer* mirroring_layer) override;
    void RemoveReflector(Reflector* reflector) override;

    scoped_refptr<cc::ContextProvider> SharedMainThreadContextProvider() override;
    void RemoveCompositor(Compositor* compositor) override;
    bool DoesCreateTestContexts() override;
    uint32_t GetImageTextureTarget(gfx::BufferFormat format,
        gfx::BufferUsage usage) override;
    cc::SharedBitmapManager* GetSharedBitmapManager() override;
    gpu::GpuMemoryBufferManager* GetGpuMemoryBufferManager() override;
    cc::TaskGraphRunner* GetTaskGraphRunner() override;
    scoped_ptr<cc::SurfaceIdAllocator> CreateSurfaceIdAllocator() override;
    void ResizeDisplay(ui::Compositor* compositor,
        const gfx::Size& size) override;

private:
    scoped_refptr<InProcessContextProvider> shared_main_thread_contexts_;
    scoped_refptr<InProcessContextProvider> shared_worker_context_provider_;
    cc::TestSharedBitmapManager shared_bitmap_manager_;
    cc::TestGpuMemoryBufferManager gpu_memory_buffer_manager_;
    cc::TestImageFactory image_factory_;
    cc::TestTaskGraphRunner task_graph_runner_;
    uint32_t next_surface_id_namespace_;
    bool use_test_surface_;
    bool context_factory_for_test_;
    cc::SurfaceManager* surface_manager_;

    base::hash_map<Compositor*, cc::OnscreenDisplayClient*> per_compositor_data_;

    DISALLOW_COPY_AND_ASSIGN(InProcessContextFactory);
};

} // namespace ui

#endif // UI_COMPOSITOR_TEST_IN_PROCESS_CONTEXT_FACTORY_H_
